package com.jingbit.ai.helper;

import android.text.TextUtils;

import java.util.regex.Pattern;

public class GetFirstLetter {

	// 字母Z使用了两个标签，这里有２７个值
	// i, u, v都不做声母, 跟随前面的字母
	private static final char[] CHARS = { '啊', '芭', '擦', '搭', '蛾', '发', '噶', '哈', '哈',
			'击', '喀', '垃', '妈', '拿', '哦', '啪', '期', '然', '撒', '塌', '塌', '塌','挖', '昔', '压', '匝', '座' };

	private static final char[] ALPHAS = { 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I',
			'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V','W', 'X', 'Y', 'Z' };

	private static int[] table = new int[27];
	// 初始化
	static{
		for (int i = 0; i < 27; ++i) {
			table[i] = gbValue(CHARS[i]);
		}
	}

	// 主函数,输入字符,得到他的声母,
	// 英文字母返回对应的大写字母
	// 其他非简体汉字返回 '0'

	public static char char2Alpha(char ch) {

		if (ch >= 'a' && ch <= 'z')
			return (char) (ch - 'a' + 'A');
		if (ch >= 'A' && ch <= 'Z')
			return ch;
		int gb = gbValue(ch);
		if (gb < table[0])
			return '#';
		int i;
		for (i = 0; i < 26; ++i) {
			if (match(i, gb))
				break;
		}
		if (i >= 26)
			return '#';
		else
			return ALPHAS[i];
	}

	// 根据一个包含汉字的字符串返回一个汉字拼音首字母的字符串
	public static String getLetter(String sourceStr) {
		try {
			if (sourceStr.startsWith("婀")) {
				return "E";
			}
			else if (sourceStr.startsWith("1")) {
				return "Y";
			}
			else if (sourceStr.startsWith("2")) {
				return "E";
			}
			else if (sourceStr.startsWith("3")) {
				return "S";
			}
			else if (sourceStr.startsWith("4")) {
				return "S";
			}
			else if (sourceStr.startsWith("5")) {
				return "W";
			}
			else if (sourceStr.startsWith("6")) {
				return "L";
			}
			else if (sourceStr.startsWith("7")) {
				return "Q";
			}
			else if (sourceStr.startsWith("8")) {
				return "B";
			}
			else if (sourceStr.startsWith("9")) {
				return "J";
			}
			return char2Alpha(sourceStr.charAt(0))+"";
		} catch (Exception e) {
			return "#";
		}
	}

	private static boolean match(int i, int gb) {
		if (gb < table[i])
			return false;
		int j = i + 1;

		// 字母Z使用了两个标签
		while (j < 26 && (table[j] == table[i]))
			++j;
		if (j == 26)
			return gb <= table[j];
		else
			return gb < table[j];
	}

	// 取出汉字的编码
	private static int gbValue(char ch) {
		String str = new String();
		str += ch;
		try {
			byte[] bytes = str.getBytes("GB2312");
			if (bytes.length < 2)
				return 0;
			return (bytes[0] << 8 & 0xff00) + (bytes[1] & 0xff);
		} catch (Exception e) {
			return 0;
		}
	}
	
	  /**
     * 获取拼音的首字母（大写）
     * @param pinyin
     * @return
     */
    public static String getFirstLetter(final String pinyin){
        if (TextUtils.isEmpty(pinyin)) return "";
        String c = pinyin.substring(0, 1);
        Pattern pattern = Pattern.compile("^[A-Za-z]+$");
        if (pattern.matcher(c).matches()){
            return c.toUpperCase();
        } else if ("0".equals(c)){
            return "";
        } else if ("1".equals(c)){
            return "热门";
        }
        return "";
    }
}